#! /usr/bin/env bash
#
#  Simple pre-commit hook script, enforcing clang-format on our tree
#
#  Licensed to the Apache Software Foundation (ASF) under one
#  or more contributor license agreements.  See the NOTICE file
#  distributed with this work for additional information
#  regarding copyright ownership.  The ASF licenses this file
#  to you under the Apache License, Version 2.0 (the
#  "License"); you may not use this file except in compliance
#  with the License.  You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software
#  distributed under the License is distributed on an "AS IS" BASIS,
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#  See the License for the specific language governing permissions and
#  limitations under the License.

GIT_TOP="$(git rev-parse --show-toplevel)"
source "$GIT_TOP/tools/clang-format.sh"

case $(uname -s) in
Darwin)
    FORMAT=${FORMAT:-${ROOT}/clang-format/clang-format.macos.$(uname -m)}
    ;;
Linux)
    FORMAT=${FORMAT:-${ROOT}/clang-format/clang-format.linux.$(uname -m)}
    ;;
*)
    echo "Leif needs to build a clang-format for $(uname -s)"
    exit 2
    ;;
esac

# If there is no clang-format in our git repo, then try from git config
[ ! -x "$FORMAT" ] && FORMAT=$(git config clangFormat.binary)

# Make sure we have some clang-format executable...
if [ ! -x "$FORMAT" ]; then
    echo "No clang-format found"
    exit 1
fi

source "$GIT_TOP/tools/yapf.sh"
if [ ! -d ${YAPF_VENV} ]; then
    echo "Run \"cmake --build <build_dir> --target yapf\""
    exit 1
fi

source "$GIT_TOP/tools/cmake-format.sh"
if [ ! -d ${CMAKE_FORMAT_VENV} ]; then
    echo "Run \"cmake --build <build_dir> --target cmake-format\""
    exit 1
fi

# Where to store the patch
clang_patch_file=$(mktemp -t clang-format.XXXXXXXXXX)
yapf_patch_file=$(mktemp -t yapf.XXXXXXXXXX)
cmake_format_patch_file=$(mktemp -t cmake-format.XXXXXXXXXX)
trap "rm -f $clang_patch_file $yapf_patch_file $cmake_format_patch_file" 0 1 2 3 5 15

# Loop over all files that are changed, and produce a diff file
source ${YAPF_VENV}/bin/activate
REPO_ROOT=$(cd $(dirname $0)/../.. && git rev-parse --show-toplevel)
YAPF_CONFIG=${REPO_ROOT}/.style.yapf
git diff-index --cached --diff-filter=ACMR --name-only HEAD | grep -vE "lib/(Catch2|fastlz|ls-hpack|swoc|yamlcpp)" |  while read file; do
    case "$file" in
    *.cc | *.c | *.h | *.h.in)
        ${FORMAT} "$file" | diff -u "$file" - >>"$clang_patch_file"
        ;;
    # Keep this list of Python extensions the same with the list of
    # extensions searched for in the toosl/yapf.sh script.
    *.py | *.cli.ext | *.test.ext)
        yapf \
            --style ${YAPF_CONFIG} \
            --parallel \
            --diff \
            "$file" >>"$yapf_patch_file"
        ;;
    esac
done
deactivate

# Now repeat the above for CMakeLists.txt files.
source ${CMAKE_FORMAT_VENV}/bin/activate
git diff-index --cached --diff-filter=ACMR --name-only HEAD | grep -E 'CMakeLists.txt|.cmake$' | grep -vE "lib/(Catch2|fastlz|ls-hpack|swoc|yamlcpp)" | while read file; do
    cmake-format "$file" | diff -u "$file" - >>"$cmake_format_patch_file"
done
deactivate

if [ -s "$clang_patch_file" ]; then
    echo "The commit is not accepted, because clang-format does not match current"
    echo "requirements. Easiest to fix this is to run:"
    echo
    echo "    $ cmake --build <build_dir> --target format"
    exit 1
else
    echo "This commit complies with the current clang-format indentation rules."
    echo
fi

if [ -s "$yapf_patch_file" ]; then
    echo "The commit is not accepted because yapf reports issues with it."
    echo "The easiest way to fix this is to run:"
    echo
    echo "    $ cmake --build <build_dir> --target format"
    exit 1
else
    echo "This commit complies with the current yapf formatting rules."
    echo
fi

if [ -s "$cmake_format_patch_file" ]; then
    echo "The commit is not accepted because cmake-format reports issues with it."
    echo "The easiest way to fix this is to run:"
    echo
    echo "    $ cmake --build <build_dir> --target format"
    exit 1
else
    echo "This commit complies with the current cmake-format formatting rules."
    echo
fi

# Cleanup before exit
rm -f "$clang_patch_file" "$yapf_patch_file"
exit 0
